#设置模板：Setting->FaCT->Python Script
#文件存放目录：D:/Python/Tprogram
import os

#定义变量名filename
filename = 'student.txt'

#定义主函数
def main():
    while True:
        menm()
        choice = int(input('请选择：'))
        if choice in [0,1,2,3,4,5,6,7]:
            if choice == 0:
                answer = input('您确定要退出系统么？y/n')
                if answer == 'y' or answer == 'Y':
                    print('感谢您的使用！！！')
                    break
                else:
                    continue
            elif choice == 1:
                insert()    #录入学生信息
            elif choice == 2:
                search()    #查询学生信息
            elif choice == 3:
                delete()
            elif choice == 4:
                modify()
            elif choice == 5:
                sort()
            elif choice == 6:
                total()
            elif choice == 7:
                show()

#定义主菜单
def menm():
    print('====================学生信息管理系统=====================')
    print('-----------------------功能菜单------------------------')
    print('\t\t\t\t\t1.录入学生信息')
    print('\t\t\t\t\t2.查找学生信息')
    print('\t\t\t\t\t3.删除学生信息')
    print('\t\t\t\t\t4.修改学生信息')
    print('\t\t\t\t\t5.排序')
    print('\t\t\t\t\t6.统计学生总人数')
    print('\t\t\t\t\t7.显示所有学生信息')
    print('\t\t\t\t\t0.退出系统')
    print('=======================================================')

#主菜单函数定义
def insert():
    student_list = []
    while True:
        id = input('请输入id：')
        if not id:      #not id 如果id不是0，就继续执行函数
                        #知识点非0数字、字符串的布尔值为True，0和空字符串的布尔值为False
            break
        name = input('请输入姓名：')
        if not name:
            break

        try:
            englist = int(input('请输入英语成绩：'))
            python = int(input('请输入Python成绩：'))
            java = int(input('请输入Java成绩：'))
        except:
            print('输入无效，请输入整数类型的成绩')
            continue
        #将录入的学生信息保存到字典当中
        student = {'id':id,'name':name,'english':englist,'python':python,'java':java}
        #将学生信息添加到列表中
        student_list.append(student)
        answer = input('是否继续添加？y/n\n')
        if answer == 'y' or answer == 'Y':
            continue
        else:
            break

    #将信息添加完，之后要把信息保存到文件中，调用save()函数
    save(student_list)
    print('学生信息录入完毕！！！')

#定义save函数实现将列表存到磁盘上
#新建一个变量名 用于存放列表信息 写在最上面
def save(lst):
    #保存的时候可能会出现异常
    try:
        stu_txt = open(filename,'a',encoding='utf-8')       #a 表示追加
    except:
        #如果没有文件，采用只写的模式打开文件
        stu_txt = open(filename,'w',encoding='utf-8')
    #遍历list
    for item in lst:
        stu_txt.write(str(item)+'\n')
    stu_txt.close()     #别忘了close操作

def search():
    student_query = []
    while True:
        id = ''
        name = ''
        #查找学生信息
        if os.path.exists(filename):
            mode = input("按照ID查找请输入1，按照姓名查找请输入2：")
            if mode == '1':
                id = input("请输入想要查询学生的id：")
            elif mode == '2':
                name = input("请输入想要查询学生的姓名：")
            else:
                print("您的输入有误，请重新输入：")
                search()
            with open(filename,'r',encoding='utf-8') as rfile:
                student = rfile.readlines()
                for item in student:
                    d = dict(eval(item))
                    if id != '':
                        if d['id'] == id:
                            student_query.append(d)
                    elif name != '':
                        if d['name'] == name:
                            student_query.append(d)

            #显示查找结果
            show_student(student_query)
            #每次查找之后都要清空查找队列
            student_query.clear()
            #是否继续
            answer = input("是否要继续查询？y/n:")
            if answer == 'y':
                continue
            else:
                break

        else:
            print("暂未保存学生信息")
            return

def show_student(lst):
    if len(lst) == 0:
        print('没有查询到学生信息，无数据显示！')
        return
    #定义标题格式：
    format_title = '{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^8}'
    print(format_title.format('ID','姓名','英语成绩','Python成绩','Java成绩','总成绩'))
    #定义内容格式：
    format_date = '{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^8}'
    for item in lst:
        print(format_date.format(
            item.get('id'),
            item.get('name'),
            item.get('english'),
            item.get('python'),
            item.get('java'),
            int(item.get('english'))+int(item.get('python'))+int(item.get('java'))
        ))

#实现信息删除功能
def delete():
    while True:
        student_id = input('请输入要删除学生的id：')
        if student_id != '':    #如果id不为空 和上面insert的not一个意思
            # 判断磁盘上文件是否存在，进行os操作，要导入os的内置模块
            if os.path.exists(filename):
                with open(filename, 'r', encoding='utf-8') as file:  # 使用上下文管理器以只读的方式打开文件
                    student_old = file.readlines()  # 这里是readlines
            else:
                student_old = []
            flag = False  # 设置flag，用来标记是否删除，默认不删除
            if student_old:
                with open(filename, 'w', encoding='utf-8') as wfile:
                    d = {}
                    for item in student_old:
                        d = dict(eval(item))    #将字符串转成字典
                        if d['id'] != student_id:
                            wfile.write(str(d)+'\n')
                        else:
                            flag = True

                    if flag:
                        print(f'id为{student_id}的学生信息已被删除')      #f 的作用是格式化紫福春，使引号中的变量或表达式能正常使用
                    else:
                        print(f'id为{student_id}的学生信息未被找到')
            else:
                print('无学生信息')
                break
            show()      #删除时候要重新显示所有的学生信息
            answer = input('是否继续删除？y/n\n')
            if answer == 'y' or answer == 'Y':
                continue
            else:
                break
    print('学生信息删除完毕！！！')

def modify():
    show()
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8') as rfile:
            student_old = rfile.readlines()
    else:
        return
    student_id = input('请输入要修改的学员的id：')
    with open(filename,'w',encoding='utf-8') as wfile:
        for item in student_old:
            d = dict(eval(item))
            if d['id'] == student_id:
                print('找到学生信息，可以修改他的相关信息了！')
                while True:
                    try:
                        d['name'] = input('请输入姓名：')
                        d['english'] = input('请输入英语成绩：')
                        d['python'] = input('请输入python成绩：')
                        d['java'] = input('请输入java成绩：')
                    except:
                        print('您的输入有误，请重新输入：')
                    else:
                        break
                wfile.write(str(d)+'/n')
                print('修改成功！！！')
            else:
                wfile.write(str(d)+'\n')
        answer = input("是否需要修改其他学生信息？y/n\n")
        if answer == 'y':
            modify()

def sort():
    show()
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8') as rfile:
            student_lst = rfile.readlines()
        student_new = []
        for item in student_lst:
            d = dict(eval(item))
            student_new.append(d)
    else:
        print("文件不存在，返回主菜单")
        return
    asc_or_desc = input("请选择升序还是降序（0升序，1降序）：")
    if asc_or_desc == '0':
        asc_or_desc = False
    elif asc_or_desc == '1':
        asc_or_desc = True
    else:
        print("您的输入有误，请重新输入：")
        sort()
    mode =  input("请选择排序方式（1 按英语成绩排序，2 按Python成绩排序，3 按 Java成绩排序，0 按照总成绩排序）：")
    if mode == '1':
        student_new.sort(key=lambda x :int(x['english']),reverse=asc_or_desc)
    elif mode == '2':
        student_new.sort(key=lambda x :int(x['python']),reverse=asc_or_desc)
    elif mode == '3':
        student_new.sort(key=lambda x :int(x['java']),reverse=asc_or_desc)
    elif mode == '0':
        student_new.sort(key=lambda x :int(x['english'])+int(x['python'])+int(x['java']),reverse=asc_or_desc)
    else:
        print("您的输入有误，请重新输入！！！\n")
        sort()
    #展示学生信息
    show_student(student_new)

def total():
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8')as rfile:
            students = rfile.readlines()
            if students:
                print(f'一共有{len(students)}名同学')
            else:
                print("还没有录入学生信息")
    else:
        print("暂未保存数据信息......")

def show():
    student_lst = []
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8') as rfile:
            students = rfile.readlines()
            for item in students:
                student_lst.append(eval(item))
            if student_lst:
                show_student(student_lst)
    else:
        print("暂未保存过数据...")

if __name__ == '__main__':
    main()


